package com.kevin.spider.utils;

import com.microsoft.playwright.Page;
import lombok.SneakyThrows;

import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;

/**
 * @author 张帆
 * @date 2025/6/17 9:54
 */
public class PlaywrightUtils {

    public static void writeHeader(Page page) {

    }

    @SneakyThrows
    public static void loadHeader(Page page) {
        // 从文件读取 headers
        String content = new String(Files.readAllBytes(Paths.get("headers.json")));
        // 注意：这里需要解析字符串为 Map，实际应根据保存格式调整
        Map<String, String> headers = new HashMap<>();
        // 示例：简单解析字符串转 Map，建议使用 JSON 解析库
        for (String entry : content.replaceAll("[{}]", "").split(",")) {
            String[] pair = entry.trim().split("=");
            if (pair.length == 2) {
                headers.put(pair[0], pair[1]);
            }
        }
        // 设置请求头
        page.setExtraHTTPHeaders(headers);
    }
}
